import numpy as np
#创建一个数组
data = np.arange(10)
print('原始数据：',data)

#切分数组
train,test = np.split(data,[7])
print('训练集：',train)
print('测试集：',test)
